import numpy as np
version_num='d35'
train_dt = np.load('E:\\data_pi\\test_dts_d35.npy')
fdt=[]
for dt in train_dt:
    t=np.zeros(7)
    y,m,d=dt.split('-')
    if y=='2020':
        if m=='10':
            t[0]=1
        elif m=='11':
            t[1]=1
        elif m=='12':
            t[2]=1
        else:
            print('out of bound')
            break
    elif y=='2021':
        if m=='01':
            t[3]=1
        elif m=='02':
            t[4]=1
        elif m == '03':
            t[5] = 1
        elif m == '04':
            t[6] = 1
        else:
            print('out of bound')
            break
    fdt.append(t)
np.save('E:\\data_pi\\test_dts_'+version_num+'_m.npy',fdt)